Repeat along given axis with as_strided